4-3 Keras Tutorial

发表于 | 分类于 | 阅读次数:
字数统计: 65 | 阅读时长 ≈ 1

np.expand_dims:用于扩展数组的形状

1
2
3
4
5
6
7
8
9
10
11
12
13
14
>>> x = np.array([1,2])
>>> x.shape
(2,)
>>> y = np.expand_dims(x, axis=0)
>>> y
array([[1, 2]])
>>> y.shape
(1, 2)
>>> y = np.expand_dims(x, axis=1)  # Equivalent to x[:,newaxis]
>>> y
array([[1],
       [2]])
>>> y.shape
(2, 1)